from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.ensemble import RandomForestClassifier
from sklearn.neural_network import MLPClassifier
from sklearn.tree import DecisionTreeClassifier

from factors import (
    accounting_factors,
    market_factors,
    academic_factors,
)
from single_model_helper import single_model_runner

factors = accounting_factors + market_factors + academic_factors

for i in "abcde":
    lr = RandomForestClassifier()
    print(single_model_runner(lr, f"preFD_{i}", factors))


# used:
# MLPClassifier(hidden_layer_sizes=(100, 50, 10), max_iter=100000)
# SVC(probability=True)
